package com.ruoyi.system.service.impl;

import java.util.ArrayList;
import java.util.List;
import java.util.stream.Collectors;

import com.baomidou.mybatisplus.core.conditions.query.LambdaQueryWrapper;
import com.baomidou.mybatisplus.extension.service.impl.ServiceImpl;
import com.ruoyi.common.utils.DateUtils;
import com.ruoyi.common.utils.bean.BeanUtils;
import com.ruoyi.system.domain.Qdkb;
import com.ruoyi.system.domain.Wdqd;
import com.ruoyi.system.domain.vo.PqVo;
import com.ruoyi.system.mapper.QdkbMapper;
import com.ruoyi.system.service.IWdqdService;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.stereotype.Service;
import com.ruoyi.system.mapper.PqMapper;
import com.ruoyi.system.domain.Pq;
import com.ruoyi.system.service.IPqService;
import org.springframework.transaction.annotation.Transactional;

/**
 * 【请填写功能名称】Service业务层处理
 *
 * @author ruoyi
 * @date 2025-04-09
 */
@Service
public class PqServiceImpl extends ServiceImpl<PqMapper, Pq> implements IPqService
{
    @Autowired
    private PqMapper pqMapper;
    @Autowired
    private IWdqdService wdqdService;



    /**
     * 查询【请填写功能名称】
     *
     * @param scheduleId 【请填写功能名称】主键
     * @return 【请填写功能名称】
     */
    @Override
    public Pq selectPqByScheduleId(Long scheduleId)
    {
        return pqMapper.selectPqByScheduleId(scheduleId);
    }

    @Override
    @Transactional
    public boolean pmScheduleAdd(PqVo pqVo,Pq pq) {

        pq.setPmOrTl("1");
        List<Pq> list = pqMapper.selectPqList(pq);
        List<Long> scheduleIdList=list.stream().map(v->v.getScheduleId()).collect(Collectors.toList());
        this.removeBatchByIds(scheduleIdList);
        String[] us = pqVo.getUserIdsString().split("_");
        String[] os = pqVo.getOrderIdsString().split("_");
        List<Pq> list1=new ArrayList<>();
        for (int i=0;i<us.length;i++){
            Pq pq1=new Pq();
            pq1.setOrderId(Long.valueOf(os[i]));
            pq1.setQdkbId(pqVo.getQdkbId());
            pq1.setUserId(Long.valueOf(us[i]));
            pq1.setPmOrTl("1");
            pq1.setScheduleStartDate(pqVo.getScheduleStartDate());
            pq1.setScheduleEndDate(pqVo.getScheduleEndDate());
            list1.add(pq1);
        }
        return this.saveBatch(list1);
    }

    @Override
    @Transactional
    public boolean tlScheduleAdd(PqVo pqVo, Pq pq) {
        pq.setPmOrTl("2");
        List<Pq> list = pqMapper.selectPqList(pq);
        List<Long> scheduleIdList=list.stream().map(v->v.getScheduleId()).collect(Collectors.toList());
        this.removeBatchByIds(scheduleIdList);
        String[] us = pqVo.getUserIdsString().split("_");
        List<Pq> list1=new ArrayList<>();
        LambdaQueryWrapper<Wdqd> lambdaQueryWrapper=new LambdaQueryWrapper<>();
        lambdaQueryWrapper.select(Wdqd::getScheduleStatus,Wdqd::getOrderId);
        lambdaQueryWrapper.eq(Wdqd::getOrderId,pqVo.getOrderId());
        lambdaQueryWrapper.eq(Wdqd::getScheduleStatus,"2");
        Wdqd wdqd = wdqdService.getOne(lambdaQueryWrapper);
        if(wdqd!=null){
            wdqd.setScheduleStatus("1");
            wdqdService.updateById(wdqd);
        }
        for (int i=0;i<us.length;i++){
            Pq pq1=new Pq();
            pq1.setOrderId(pqVo.getOrderId());
            pq1.setQdkbId(pqVo.getQdkbId());
            pq1.setUserId(Long.valueOf(us[i]));
            pq1.setPmOrTl("2");
            pq1.setScheduleStartDate(pqVo.getScheduleStartDate());
            pq1.setScheduleEndDate(pqVo.getScheduleEndDate());
            list1.add(pq1);
        }
        return this.saveBatch(list1);
    }

    /**
     * 查询【请填写功能名称】列表
     *
     * @param pq 【请填写功能名称】
     * @return 【请填写功能名称】
     */
    @Override
    public List<Pq> selectPqList(Pq pq)
    {
        return pqMapper.selectPqList(pq);
    }

    /**
     * 新增【请填写功能名称】
     *
     * @param pq 【请填写功能名称】
     * @return 结果
     */
    @Override
    public int insertPq(Pq pq)
    {
        pq.setCreateTime(DateUtils.getNowDate());
        return pqMapper.insertPq(pq);
    }

    /**
     * 修改【请填写功能名称】
     *
     * @param pq 【请填写功能名称】
     * @return 结果
     */
    @Override
    public int updatePq(Pq pq)
    {
        return pqMapper.updatePq(pq);
    }

    /**
     * 批量删除【请填写功能名称】
     *
     * @param scheduleIds 需要删除的【请填写功能名称】主键
     * @return 结果
     */
    @Override
    public int deletePqByScheduleIds(Long[] scheduleIds)
    {
        return pqMapper.deletePqByScheduleIds(scheduleIds);
    }

    /**
     * 删除【请填写功能名称】信息
     *
     * @param scheduleId 【请填写功能名称】主键
     * @return 结果
     */
    @Override
    public int deletePqByScheduleId(Long scheduleId)
    {
        return pqMapper.deletePqByScheduleId(scheduleId);
    }
}
